import { Message } from "element-ui";
function downloadFile(url, fileName = "download") {
  return fetch(url)
    .then((response) => response.blob())
    .then((blob) => {
      const url = window.URL.createObjectURL(blob);
      const a = document.createElement("a");
      a.href = url;
      a.download = fileName; // 这里可以替换为你想要保存的文件名和扩展名
      document.body.appendChild(a);
      a.click();
      window.URL.revokeObjectURL(url);
    })
    .catch((error) => {
      Message.error("下载失败");
    });
}

import FileSaver from "file-saver";
import JSZip from 'jszip'

function streamToBlob(readableStream) {
  return new Response(readableStream).blob();
}

// 批量打包下载
// fileArr下载的list：例如：
export const downloadZip = async (fileArr, zipName = '下载文件') => {
  fileArr.forEach(el => {
    el.fvName = `${el.originName}.${el.file.split('.').at(-1)}`  // 截取url的后缀名添加到文件名： fvname.jpg
  })
  const zip = new JSZip() // 创建实例对象
  const promises = []
  // 遍历生成下载文件
  fileArr.forEach((item) => {
    const promise = getFile(item.file).then(async (res) => {

      let blob = await streamToBlob(res.body);

      const fvName = item.fvName + ''
      // 创建文件用file()，创建文件夹用 floder()
      zip.file(fvName, blob, {
        binary: true
      });

      console.log("res", res);
    
    })
    promises.push(promise)
  })

  // 生成 zip 文件
  Promise.all(promises).then(() => {
    // 生成zip 文件
    zip.generateAsync({
      type: 'blob',  // 文件格式
      compression: 'DEFLATE', // STORE: 默认不压缩， DEFLATE：需要压缩
      compressionOptions: {
        level: 9 // 压缩等级 1~9   1 压缩速度最快， 9 最优压缩方式
      }
    }).then((res) => {
      FileSaver.saveAs(res, `${zipName}.zip`) // 使用FileSaver.saveAs保存文件，文件名可自定义
    })
  })
}

// 这一步为下载文件步骤，通过传入的url下载文件
// 将文件 url 格式转换为 Bolb 类型格式 或者 arraybuffer 格式
const getFile = (fvUrl) => {
  
  return new Promise((resolve, reject) => {
    fetch(fvUrl, {
      method: 'get',  // get请求
      responseType: 'blob' // 返回的数据会被强制转为blob类型 ，转换成arraybuffer 也行
    }).then((res) => {
      resolve(res) // 将下载的文件返回
    }).catch(err => {
      reject(err)
    })
  })
}

export const download = (URL, fileName) => {
  const oReq = new XMLHttpRequest();
  oReq.open("GET", URL, true);
  oReq.responseType = "blob";
  oReq.onload = function (e) {
    const disposition = oReq.getResponseHeader("Content-Disposition");
    let actualFileName =
      fileName || getFileNameFromDisposition(disposition) || "downloaded_file"; // 优先使用传入的文件名，其次是响应头中的文件名，最后默认名称

    const file = new Blob([oReq.response], {
      type: oReq.getResponseHeader("Content-Type"),
    });
    FileSaver.saveAs(file, actualFileName);
  };
  oReq.onerror = function (e) {
    // 在这里处理下载失败的情况
    console.error("下载失败:", e);
    Message.error("下载失败");
  };
  oReq.send();
};

const getFileNameFromDisposition = (disposition) => {
  console.log(disposition);
  if (disposition && disposition.indexOf("attachment") !== -1) {
    const fileNameRegex = /filename[^;=\n]*=((['"]).*?\2|[^;\n]*)/;
    const matches = fileNameRegex.exec(disposition);
    if (matches != null && matches[1]) {
      return matches[1].replace(/['"]/g, "");
    }
  }
  return null;
};

export { downloadFile };
